package cn.edu.xjtu.carlWay.dynamicProgramming.perfectSquares;

import java.util.Arrays;

/**
 * 279. 完全平方数
 * 给你一个整数 n ，返回 和为 n 的完全平方数的最少数量 。
 * <p>
 * 完全平方数 是一个整数，其值等于另一个整数的平方；换句话说，其值等于一个整数自乘的积。例如，1、4、9 和 16 都是完全平方数，而 3 和 11 不是。
 * <p>
 * https://leetcode-cn.com/problems/perfect-squares/
 */
public class Solution {
    public static void main(String[] args) {
        Solution solution = new Solution();
        System.out.println(solution.numSquares(12));
    }

    public int numSquares(int n) {
        int[] squares = getSquares(n);
        int[] dp = new int[n + 1];
        Arrays.fill(dp, n + 1);
        dp[0] = 0;

        for (int i = 1; i < squares.length; i++) {
            for (int j = squares[i]; j < dp.length; j++) {
                dp[j] = Math.min(dp[j], dp[j - squares[i]] + 1);
            }
        }
        return dp[n];
    }

    private int[] getSquares(int n) {
        int length = (int) Math.sqrt(n);
        int[] list = new int[length + 1];
        for (int i = 1; i < list.length; i++) {
            list[i] = i * i;
        }
        return list;
    }
}
